src.air_data_generation

 1"""
 2This file generates simulated pollution data over a 24 hour period.
 3Output saved as a JSON file, structured to mimic the UTMC Open Data Service
 4and to be consumed by PANT Spring API
 5Author: Ross Cochrane
 6"""
 7
 8import random
 9import json
10from datetime import datetime, timedelta
11import os
12
13
14
15# Settings for the simulation:
16num_sites = 130
17start_time = datetime(2025, 5, 19, 0, 0, 0)
18interval_minutes = 10
19num_points = (24 * 60) // interval_minutes  # 144 points for 24 hours
20
21
22def is_busy_period(ts):
23    """
24    A function to determine if the timestamp is during a busy period 
25    ie 8:00-9:00 or 16:00-19:00.
26    """
27    hour = ts.hour
28    return (8 <= hour < 9) or (16 <= hour < 19)
29
30
31
32# A program to generate the dataset.
33sites = []
34for site_num in range(1, num_sites + 1):
35    system_code = f"SITE{site_num:03d}"
36    dynamics = []
37
38    for i in range(num_points):
39        current_time = start_time + timedelta(minutes=i * interval_minutes)
40        busy = is_busy_period(current_time)
41
42        
43        # Simulate pollutant and noise values based on busy or quiet period.
44        # Data points taken from: no and no2: Sunderland Wessington Way. co: Newcastle 
45        # centre historic (2012) monitoring sites
46        # Noise: www.extrium.co.uk/noiseviewer.html
47        
48        co = round(random.uniform(0.5, 5.0) if busy else random.uniform(0.1, 0.17), 2)
49        no = round(random.uniform(20, 150) if busy else random.uniform(1, 10), 2)
50        no2 = round(random.uniform(40, 300) if busy else random.uniform(5, 30), 2)
51        noise = round(random.uniform(70, 100) if busy else random.uniform(30, 60), 2)
52        rh = round(random.uniform(70, 85) if busy else random.uniform(55, 59.9))
53        
54        # Set the temperature according to the time of day
55        # TO DO: Change this to follow a day pattern
56        if 10 <= current_time.hour < 18:
57            temperature = round(random.uniform(18, 24), 1)
58        elif 22 <= current_time.hour or current_time.hour < 6:
59            temperature = round(random.uniform(5, 10), 1)
60        else:
61            temperature = round(random.uniform(7, 17), 1)
62
63        battery = round(random.uniform(3.5, 4.2), 1)
64
65        dynamics.append({
66            "co": co,
67            "no": no,
68            "no2": no2,
69            "rh": rh,
70            "temperature": temperature,
71            "noise": noise,
72            "battery": battery,
73            "lastUpdated": current_time.strftime("%Y-%m-%dT%H:%M:%S.000+0000")
74        })
75
76    sites.append({
77        "systemCodeNumber": system_code,
78        "dynamics": dynamics
79    })
80
81
82# Save JSON to data folder
83
84os.makedirs("data", exist_ok=True)
85
86file_path = os.path.join("data", "pollution_data.json")
87with open(file_path, "w") as json_file:
88    json.dump(sites, json_file, indent=4)
89
90print(f"JSON data successfully saved to {file_path}")